# Copyright (c) OpenMMLab. All rights reserved.
def multi_turn_1_map_fn(example):
    messages = example["messages"]
    conversation = []
    for msg in messages:
        conversation.append(
            {
                "system": msg["toy_system"],
                "input": msg["toy_input"],
                "output": msg["toy_output"],
            }
        )
    return {"conversation": conversation}
